#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/delay.h>
#include <linux/ide.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/gpio.h>
#include <linux/cdev.h>
#include <linux/device.h>
#include <linux/of.h>
#include <linux/of_address.h>
#include <linux/platform_device.h>
#include <asm/mach/map.h>
#include <asm/uaccess.h>
#include <asm/io.h>

#define LEDDEV_CNT		1			/* 设备号长度 	*/
#define LEDDEV_NAME		"platled"	/* 设备名字 	*/
#define LEDOFF 			0
#define LEDON 			1

/* 寄存器名 */
static void __iomem *IMX6U_CCM_CCGR1;
static void __iomem *SW_MUX_GPIO1_IO03;
static void __iomem *SW_PAD_GPIO1_IO03;
static void __iomem *GPIO1_DR;
static void __iomem *GPIO1_GDIR;

/* leddev设备结构体 */
struct leddev_dev{
	dev_t devid;			/* 设备号	*/
	struct cdev cdev;		/* cdev		*/
	struct class *class;	/* 类 		*/
	struct device *device;	/* 设备		*/
	int major;				/* 主设备号	*/		
	int minor;              /* 次设备号	*/	
};

struct leddev_dev leddev; 	/* led设备 */

/*
 * @description		: LED打开/关闭
 * @param - sta 	: LEDON(0) 打开LED，LEDOFF(1) 关闭LED
 * @return 			: 无
 */
void led0_switch(u8 sta)
{
	u32 val = 0;
	if(sta == LEDON) 
	{
		val = readl(GPIO1_DR);
		val &= ~(1 << 3);	
		writel(val, GPIO1_DR);
	}
	else if(sta == LEDOFF) 
	{
		val = readl(GPIO1_DR);
		val|= (1 << 3);	
		writel(val, GPIO1_DR);
	}	
}

/**
  * @brief  打开设备
  * @note   
  * @param  inode 传递给驱动的inode
  * @param  filp 设备文件，file结构体有个叫做private_data的成员变量,一般在open的时候将private_data指向设备结构体。
  * @retval 0 成功;其他 失败
  */
static int led_open(struct inode *inode, struct file *filp)
{
	printk("led_open!\r\n");
	filp->private_data = &leddev; /* 设置私有数据  */
	return 0;
}

/**
  * @brief  从设备读取数据 
  * @note   
  * @param  filp 要打开的设备文件(文件描述符)
  * @param  buf 返回给用户空间的数据缓冲区
  * @param  cnt 要读取的数据长度
  * @param  offt 相对于文件首地址的偏移
  * @retval 读取的字节数，如果为负值，表示读取失败
  */
static ssize_t led_read(struct file *filp, char __user *buf, size_t cnt, loff_t *offt)
{
	printk("led_read!\r\n");
	return 0;
}

/**
  * @brief  向设备写数据 
  * @note   
  * @param  filp 设备文件，表示打开的文件描述符
  * @param  buf 要写给设备写入的数据
  * @param  cnt 要写入的数据长度
  * @param  offt 相对于文件首地址的偏移
  * @retval 写入的字节数，如果为负值，表示写入失败
  */
static ssize_t led_write(struct file *filp, const char __user *buf, size_t cnt, loff_t *offt)
{
	int retvalue;
	unsigned char databuf[1];
	unsigned char ledstat;

	retvalue = copy_from_user(databuf, buf, cnt);
	if(retvalue < 0) {
		printk("kernel write failed!\r\n");
		return -EFAULT;
	}

	ledstat = databuf[0];		/* 获取状态值 */
	printk("led_write data:%c\r\n", ledstat);
	if(ledstat == LEDON) {	
		led0_switch(LEDON);		/* 打开LED灯 */
	}else if(ledstat == LEDOFF) {
		led0_switch(LEDOFF);	/* 关闭LED灯 */
	}
	return 0;
}

/* 设备操作函数 */
static struct file_operations led_fops = {
	.owner = THIS_MODULE,
	.open = led_open,
	.read = led_read,
	.write = led_write,
};

/**
  * @brief  flatform驱动的probe函数，当驱动与,设备匹配以后此函数就会执行
  * @note   
  * @param  dev platform设备
  * @retval 0，成功;其他负值,失败
  */
static int led_probe(struct platform_device *dev)
{	
	int i = 0;
	int ressize[5];
	u32 val = 0;
	struct resource *ledsource[5];

	printk("led_probe led driver and device has matched!\r\n");
	/* 1、获取资源 */
	for (i = 0; i < 5; i++) {
		ledsource[i] = platform_get_resource(dev, IORESOURCE_MEM, i); /* 依次MEM类型资源 */
		if (!ledsource[i]) {
			dev_err(&dev->dev, "No MEM resource for always on\n");
			return -ENXIO;
		}
		ressize[i] = resource_size(ledsource[i]);	
	}	

	/* 初始化LED */
	/* 1、寄存器地址映射 */
 	IMX6U_CCM_CCGR1 = ioremap(ledsource[0]->start, ressize[0]);
	SW_MUX_GPIO1_IO03 = ioremap(ledsource[1]->start, ressize[1]);
  	SW_PAD_GPIO1_IO03 = ioremap(ledsource[2]->start, ressize[2]);
	GPIO1_DR = ioremap(ledsource[3]->start, ressize[3]);
	GPIO1_GDIR = ioremap(ledsource[4]->start, ressize[4]);

	/* 2、使能GPIO1时钟 */
	val = readl(IMX6U_CCM_CCGR1);
	val &= ~(3 << 26);				/* 清除以前的设置 */
	val |= (3 << 26);				/* 设置新值 */
	writel(val, IMX6U_CCM_CCGR1);

	/* 3、设置GPIO1_IO03的复用功能，将其复用为GPIO1_IO03，最后设置IO属性。*/
	writel(5, SW_MUX_GPIO1_IO03);
	
	/*寄存器SW_PAD_GPIO1_IO03设置IO属性
	 *bit 16:0 HYS关闭
	 *bit [15:14]: 00 默认下拉
     *bit [13]: 0 kepper功能
     *bit [12]: 1 pull/keeper使能
     *bit [11]: 0 关闭开路输出
     *bit [7:6]: 10 速度100Mhz
     *bit [5:3]: 110 R0/6驱动能力
     *bit [0]: 0 低转换率
	 */
	writel(0x10B0, SW_PAD_GPIO1_IO03);

	/* 4、设置GPIO1_IO03为输出功能 */
	val = readl(GPIO1_GDIR);
	val &= ~(1 << 3);	/* 清除以前的设置 */
	val |= (1 << 3);	/* 设置为输出 */
	writel(val, GPIO1_GDIR);

	/* 5、默认关闭LED */
	val = readl(GPIO1_DR);
	val |= (1 << 3);	
	writel(val, GPIO1_DR);

	/* 注册字符设备驱动 */
	/*1、创建设备号 */
	if (leddev.major) {		/*  定义了设备号 */
		leddev.devid = MKDEV(leddev.major, 0);
		register_chrdev_region(leddev.devid, LEDDEV_CNT, LEDDEV_NAME);
	} else {						/* 没有定义设备号 */
		alloc_chrdev_region(&leddev.devid, 0, LEDDEV_CNT, LEDDEV_NAME);	/* 申请设备号 */
		leddev.major = MAJOR(leddev.devid);	/* 获取分配号的主设备号 */
		leddev.minor = MINOR(leddev.devid);	/* 获取分配号的次设备号 */
	}
	printk("platled major=%d,minor=%d\r\n",leddev.major, leddev.minor);	
	
	/* 2、初始化cdev */
	leddev.cdev.owner = THIS_MODULE;
	cdev_init(&leddev.cdev, &led_fops);
	
	/* 3、添加一个cdev */
	cdev_add(&leddev.cdev, leddev.devid, LEDDEV_CNT);

	/* 4、创建类 */
	leddev.class = class_create(THIS_MODULE, LEDDEV_NAME);
	if (IS_ERR(leddev.class)) {
		return PTR_ERR(leddev.class);
	}

	/* 5、创建设备 */
	leddev.device = device_create(leddev.class, NULL, leddev.devid, NULL, LEDDEV_NAME);
	if (IS_ERR(leddev.device)) {
		return PTR_ERR(leddev.device);
	}
	printk("device node: /dev/%s\r\n", LEDDEV_NAME);
	return 0;
}

/**
  * @brief  platform驱动的remove函数，移除platform驱动的时候此函数会执行
  * @note   
  * @param  dev platform设备
  * @retval 0，成功;其他负值,失败
  */
static int led_remove(struct platform_device *dev)
{
	printk("led_remove!\r\n");
	/* 取消映射 */
	iounmap(IMX6U_CCM_CCGR1);
	iounmap(SW_MUX_GPIO1_IO03);
	iounmap(SW_PAD_GPIO1_IO03);
	iounmap(GPIO1_DR);
	iounmap(GPIO1_GDIR);

	/* 注销字符设备驱动 */
	cdev_del(&leddev.cdev);/*  删除cdev */
	unregister_chrdev_region(leddev.devid, LEDDEV_CNT); /* 注销设备号 */

	device_destroy(leddev.class, leddev.devid);
	class_destroy(leddev.class);
	return 0;
}

/** 
 *  定义匹配ID数组 
 *  与数组中含有的名称相同的设备都可以匹配到该驱动
 *  结构体变量名赋值给 struct platform_driver 结构体的 .id_table 成员
 */
struct platform_device_id leddevice_ids[] = 
{
	[0] = {.name = "imx6ul-led"},
	[1] = {.name = "imx6ul-led-device"},
	[2] = {}                   /* means ending */
};

/* platform驱动结构体 */
static struct platform_driver led_driver = {
	.driver		= {
		.name	= "imx6ul-led-driver",/* 驱动名字 */
	},
	.probe		= led_probe,
	.remove		= led_remove,
	.id_table   = leddevice_ids,  /* 可匹配的设备ID数组 */
};

/**
  * @brief  驱动模块加载函数
  * @note   
  * @param  
  * @retval 
  */
static int __init leddriver_init(void)
{
	printk("leddriver_init module load!\r\n");
	return platform_driver_register(&led_driver);
}
/**
  * @brief  驱动模块卸载函数
  * @note   
  * @param  
  * @retval 
  */
static void __exit leddriver_exit(void)
{
	printk("leddriver_exit module unload!\r\n");
	platform_driver_unregister(&led_driver);
}
module_init(leddriver_init);
module_exit(leddriver_exit);
MODULE_LICENSE("GPL");
MODULE_AUTHOR("ssq");